#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os
def get_base_file_path():
    return os.path.dirname(os.path.dirname(__file__))

class Config:
    llm_api_key = 'sk-...'  # 修改为您实际的api key
    llm_model_low = 'qwen3-8b'
    llm_model_normal = 'qwen3-14b'
    llm_model_high = 'qwen3-235b-a22b'
    embedding_model = 'text-embedding-v4'
    reranker_model = 'gte-rerank-v2'
    base_path = get_base_file_path()
    result_path = os.path.join(base_path,"result")
    data_path = os.path.join(base_path, "data")
    db_sqlite_url =os.path.join(data_path,'data.db')
    questions_path = os.path.join(data_path, "questions.jsonl").replace("\\", "/")
    samples_path = os.path.join(data_path, "samples.jsonl").replace("\\", "/")
    data_answer_path = os.path.join(result_path,'results.jsonl').replace("\\", "/")
    
    check_data_path = os.path.join(result_path,'check_data.log').replace("\\", "/")
    chroma_location = './chroma_db'
    collection_name = 'text_embeddings'
    sql_output_path = './sql_output'
    requestions_path = os.path.join(data_path, "requestions.jsonl").replace("\\", "/")
    embedding_batch_size = 10
    embedding_dim_size = 1024
    tables_list = ['area_items_total_day','items_total_day','accept_src_items_total_day','order_type_items_total_day','order_status_total_day','assess_items_total_day','call_overview_total_day','call_connect_total_day','sub_top_total_day','sub_accept_total_day','sub_type_total_day','sub_area_total_day','sub_warn_total_day','sub_monitor_total_day','sub_assess_total_day','item_bmh_total_day','item_callback_total_day','item_overdue_total_day','order_flow_total_day','order_full_total_day']

project_config = Config()
